import django_filters

from apps.steps.models import Steps


class StepFilter(django_filters.rest_framework.FilterSet):

    case = django_filters.CharFilter(field_name='case')
    step = django_filters.NumberFilter('step_num', lookup_expr='lt')

    class Meta:
        model = Steps
        fields = ['case', 'step']
